Single-Rollout Asynchronous Optimization for Agentic Reinforcement Learning
单次 rollout 异步优化:让 Agent 强化学习训练更稳更强
🧭 异步强化学习训练稳定化 ★ 深度解读
Tsinghua University · Z.AI
提出单次 rollout 异步 RL 策略 SAO,以 token 级双面裁剪和更快价值更新稳定训练,在智能体推理与编码任务上显著超越 GRPO
📌 工作参考价值(4 点)
- 单次 rollout 替代组采样适用于在线推荐/Agent 场景:线上只能获得单条反馈时,可以用训练好的价值模型做基线,而非依赖组内相对奖励。这直接解决了 GRPO 在实时环境中的部署困难。
- 直接使用 rollout log-probabilities 做重要性采样,简化异步训练架构:不需要维护策略的历史快照链,大幅降低工程复杂度,对推荐模型异步在线学习特别友好。
- 价值模型的冻结注意力训练策略:只在 MoE 层更新,注意力层冻结,能显著稳定 critic 训练并降低梯度范数。这对需要快速收敛的在线价值估计(如 bandit 或 contextual bandit)有参考意义。
- Skip-Observation GAE 处理 agent 轨迹:计算优势时跳过环境反馈 token,将价值估计限定在模型动作上,消除环境噪声。对于涉及工具调用或多轮对话的推荐 agent(如对话式推荐),可借鉴该方式只对模型可控部分进行信用分配。